package leetCode.offer55_ii;

/**
 * 输入一棵二叉树的根节点，判断该树是不是平衡二叉树。
 * 如果某二叉树中任意节点的左右子树的深度相差不超过1，那么它就是一棵平衡二叉树。
 */
public class Solution1 implements Solution{

    @Override
    public boolean isBalanced(TreeNode root) {
        return deepSearch(root);
    }

    public boolean deepSearch(TreeNode root){
        if(root!=null){
            int[] max = new int[1];
            getDeep(root.left,0,max);
            int leftDeep = max[0];
            max[0] = 0;
            getDeep(root.right,0,max);
            int riDeep = max[0];
            max[0] = 0;
            if(leftDeep-riDeep>1||riDeep-leftDeep>1){
                return false;
            }
            return deepSearch(root.left)&&deepSearch(root.right);
        }
        return true;
    }

    public void getDeep(TreeNode root, Integer deep, int[] max){
        if(root!=null){
            deep = deep + 1;
            getDeep(root.left,deep,max);
            getDeep(root.right,deep,max);
            if(deep > max[0]) max[0] = deep;
        }
    }


}
